package com.huixuebao.after.util;


import cn.hutool.core.util.IdUtil;
import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.ObjectMetadata;
import com.aliyun.oss.model.PutObjectResult;
import lombok.extern.slf4j.Slf4j;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;

/**
 * @Desc: oss工具类
 * @Author: fatao_yu【fatao_yu@qq.com】
 * @Date: 2020-12-28 22:06
 */
@Slf4j
public class OSSClientUtil {

    /** 阿里云API的bucket名称 */
    private static final String bucketName = "huixuebao";
    /** 阿里云API的文件夹名称 */
    private static final String folder = "image/";
    /** 阿里云OSS客户端对象 */
    private static OSS ossClient;
    /** 阿里云API的内或外网域名 */
    private static String endpoint = "oss-cn-shanghai.aliyuncs.com";

    /**
     * 获取阿里云OSS客户端对象
     *
     * @return ossClient
     */
    public static OSS getOSSClient() {
        if(ossClient==null){
            /* 阿里云API的密钥Access Key ID */
            String accessKeyId = "LTAI4GKvU655QR4eZbRnvtaN";
            /* 阿里云API的密钥Access Key Secret */
            String accessKeySecret = "OUmbqSKqvo337pT8dcOqu9ZoAhoOCA";

            ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);
        }
        return ossClient;
    }

    /**
     * 上传图片至OSS
     *
     * @param file 上传文件（文件全路径如：D:\\image\\cake.jpg）
     * @return String 返回的唯一MD5数字签名
     */
    public static String uploadObject2OSS(File file) {
        String resultStr = null;
        try {
            // 以输入流的形式上传文件
            InputStream is = new FileInputStream(file);
            // 文件名
            String fileName = file.getName();
            //文件名称拼接
            if(fileName.contains(".")){
                String[] fileNames = fileName.split("\\.");
                fileName = fileNames[0] + "-" + IdUtil.simpleUUID() + "." + fileNames[fileNames.length-1];
            }else{
                fileName = fileName + "-" + IdUtil.simpleUUID();
            }
            // 文件大小
            long fileSize = file.length();
            // 创建上传Object的Metadata
            ObjectMetadata metadata = new ObjectMetadata();
            // 上传的文件的长度
            metadata.setContentLength(is.available());
            // 指定该Object被下载时的网页的缓存行为
            metadata.setCacheControl("no-cache");
            // 指定该Object下设置Header
            metadata.setHeader("Pragma", "no-cache");
            // 指定该Object被下载时的内容编码格式
            metadata.setContentEncoding("utf-8");
            // 文件的MIME，定义文件的类型及网页编码，决定浏览器将以什么形式、什么编码读取文件。如果用户没有指定则根据Key或文件名的扩展名生成，
            // 如果没有扩展名则填默认值application/octet-stream
            metadata.setContentType(getContentType(fileName));
            // 指定该Object被下载时的名称（指示MINME用户代理如何显示附加的文件，打开或下载，及文件名称）
            metadata.setContentDisposition("filename/filesize=" + fileName + "/" + fileSize + "Byte.");
            // 上传文件 (上传文件流的形式)
            PutObjectResult putResult = getOSSClient().putObject(bucketName, folder + fileName, is, metadata);
            // 解析结果
            resultStr = "https://"+bucketName+"."+endpoint+"/"+folder + fileName;
        } catch (Exception e) {
            e.printStackTrace();
            log.info("上传阿里云OSS服务器异常." + e.getMessage(), e);
        }
        return resultStr;
    }

    /**
     * 通过文件名判断并获取OSS服务文件上传时文件的contentType
     *
     * @param fileName 文件名
     * @return 文件的contentType
     */
    public static String getContentType(String fileName) {
        // 文件的后缀名
        String fileExtension = fileName.substring(fileName.lastIndexOf("."));
        if (".bmp".equalsIgnoreCase(fileExtension)) {
            return "image/bmp";
        }
        if (".gif".equalsIgnoreCase(fileExtension)) {
            return "image/gif";
        }
        if (".jpeg".equalsIgnoreCase(fileExtension) || ".jpg".equalsIgnoreCase(fileExtension)) {
            return "image/jpeg";
        }
        if (".png".equalsIgnoreCase(fileExtension)) {
            return "image/png";
        }
        if (".html".equalsIgnoreCase(fileExtension)) {
            return "text/html";
        }
        if (".txt".equalsIgnoreCase(fileExtension)) {
            return "text/plain";
        }
        if (".vsd".equalsIgnoreCase(fileExtension)) {
            return "application/vnd.visio";
        }
        if (".ppt".equalsIgnoreCase(fileExtension) || "pptx".equalsIgnoreCase(fileExtension)) {
            return "application/vnd.ms-powerpoint";
        }
        if (".doc".equalsIgnoreCase(fileExtension) || "docx".equalsIgnoreCase(fileExtension)) {
            return "application/msword";
        }
        if (".xml".equalsIgnoreCase(fileExtension)) {
            return "text/xml";
        }
        // 默认返回类型
        return "";
    }

    public static void main(String[] args) {
        String url = OSSClientUtil.uploadObject2OSS(new File("/Users/yufatao/Desktop/temp/business/01.jpg"));
        System.out.println(url);
    }
}
